package com.sxj.corejava.code13_jdbc;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author 石小俊
 */
public class Test08_多表查询操作 {

    public static void main(String[] args) {
//        List<Emp> emps = selectAll();
//        for (Emp emp : emps) {
//            System.out.println(emp);
//        }

        Dept dept = selectById(1);
        System.out.println(dept);
    }

    /**
     * 多对一查询
     * 查询所有员工以及其所属部门信息
     * @return
     */
    public static List<Emp> selectAll() {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        List<Emp> emps = new ArrayList<>();
        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append(" select e.id,e.name,e.salary,d.id 'd.id',d.name 'd.name',d.remark ")
                    .append(" from t_emp e ")
                    .append(" left join t_dept d ")
                    .append("     on e.dept_id = d.id ")
                    .toString();
            ps = conn.prepareStatement(sql);
            rs = ps.executeQuery();
            while (rs.next()) {
                // 此时的结果集映射涉及到两个对象
                // 一个属于当前员工对象,一个属于员工所属的部门对象
                // 此处的部门作为员工的属性存在
                // 因此,最终映射的所有数据本质上都是映射到员工对象中
                Emp emp = new Emp();
                emp.setId(rs.getInt("id"));
                emp.setName(rs.getString("name"));
                emp.setSalary(rs.getDouble("salary"));

                Dept dept = new Dept();
                dept.setId(rs.getInt("d.id"));
                dept.setName(rs.getString("d.name"));
                dept.setRemark(rs.getString("remark"));
                // 将部门数据赋值给员工对象中的部门属性
                emp.setDept(dept);

                // 每映射一行数据,生成一个员工对象,添加到员工集合中
                emps.add(emp);
            }
            return emps;
        } catch (SQLException e) {
            throw new DataAccessException("数据访问异常");
        } finally {
            JDBCUtil.close(conn, ps, rs);
        }
    }

    /**
     * 一对多查询
     * 根据部门编号查询部门以及该部门中的所有员工信息
     * @param id
     * @return
     */
    public static Dept selectById(Integer id) {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Dept dept = null;
        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append(" select d.id,d.name,d.remark,e.id 'e.id',e.name 'e.name',e.salary ")
                    .append(" from t_dept d ")
                    .append(" left join t_emp e ")
                    .append("     on d.id = e.dept_id ")
                    .append(" where d.id = ? ")
                    .toString();
            ps = conn.prepareStatement(sql);
            ps.setInt(1, id);
            rs = ps.executeQuery();
            while (rs.next()) {
                // 一对多的映射
                // 一方数据只映射一次
                // 在此处,一方数据表示的是部门相关的数据
                // 当dept的值为null时说明尚未映射,如果映射过了,则dept的值就不为null了
                if (dept == null) {
                    dept = new Dept();
                    dept.setId(rs.getInt("id"));
                    dept.setName(rs.getString("name"));
                    dept.setRemark(rs.getString("remark"));
                }

                // 多方数据映射多次,每一行数据对应对方数据中的一个对象
                // 在此处,多方数据表示的是员工集合
                Emp emp = new Emp();
                emp.setId(rs.getInt("e.id"));
                emp.setName(rs.getString("e.name"));
                emp.setSalary(rs.getDouble("salary"));
                // 多方数据映射完成之后,将其添加到一方数据的集合属性中
                dept.addEmp(emp);
            }
            return dept;
        } catch (SQLException e) {
            throw new DataAccessException("数据访问异常");
        } finally {
            JDBCUtil.close(conn, ps, rs);
        }
    }

}
